#!/usr/bin/env python

from douban.album import Album
import requests
import sys
import os

def mayday():
    h = """douban-album-dl album_id [location=./album]"""

    print(h)

def mkdir(path):
    if not os.path.exists(path):
        os.makedirs(path)

def get_album(album, path):
    idx = 0
    mkdir(path)
    os.chdir(path)
    for photo_url in album.photos():
        name = os.path.basename(photo_url)
        print("{}: saving {}".format(idx, name))
        r = requests.get(photo_url, stream=True)
        with open(name, "wb") as f:
            f.write(r.content)
        idx += 1
    print()
    print("saving album to {}, total {} images".format(path, idx))


if __name__ == "__main__":
    if len(sys.argv) == 1:
        mayday()
    else:
        album = Album(sys.argv[1])
        path = "./album" if len(sys.argv) == 2 else sys.argv[2]
        get_album(album, path)
